package cn.tedu.v5;

import java.io.*;
import java.net.*;

import java.util.*;


/**
 * Version 5      登陆与退出    功能：可以加在V3
 * Version  6     在线用户列表，统计上线名称
 *                消息转发---交互---服务器客户端文字
 */


public class Server {
    private ServerSocket serverSocket;
    private static final Integer PORT = 1357;
    //理解：1.服务器端发起提示---写出客户端：集合保存所有客户端
    private List<PrintWriter> allOut=new ArrayList<>();
    //1.私有化属性：存账号密码
    private Map<String,String> userDatabase;
    public Server() {
        try {
            serverSocket = new ServerSocket(PORT);
            //创建Map对象
            userDatabase=new HashMap<>();
            userDatabase.put("姜哲","123456");
            userDatabase.put("哲哲","654321");
        } catch (BindException e) {
            System.out.println("请检查端口号，被占用");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public void start() {
        try {
            while (true) {
                System.out.println("等待客户端链接");
                Socket socket = serverSocket.accept();
                new Thread(new ClientHandler(socket)).start();
            }
        } catch (IOException e) {
            System.out.println("客户端链接失效了");
        } finally {
            if (serverSocket != null && !serverSocket.isClosed()) {
                try {
                    serverSocket.close();
                } catch (IOException e) {
                    System.out.println("服务器关闭失败~~~");
                }
                System.out.println("服务器关闭成功~~Over");
            }
        }
    }

    public static void main(String[] args) {
        new Server().start();
    }


    //5.5 因为用户账号可能有很多，封装Login,传入参数判断
    private boolean Login(String user,String pass){
        return userDatabase.containsKey(user)&&userDatabase.get(user).equals(pass);
    }
    //5.6 handlerChat  封装：读写功能 ---放入线程中

    private class ClientHandler implements Runnable {
        private Socket socket;
        private String ip;
        private PrintWriter pw;
        private BufferedReader br;
        //4.线程中：确定那个客户端---赋予功能
        private String username;
        private String password;


        public ClientHandler(Socket s) {
            this.socket = s;
            ip = socket.getInetAddress().getHostAddress();
            try {
                br = new BufferedReader(new InputStreamReader(socket.getInputStream(), "UTF8"));
                pw = new PrintWriter(new BufferedWriter(new OutputStreamWriter(socket.getOutputStream(), "UTF8")), true);
                synchronized (allOut){  //监视集合中，只允许一个客户端过来
                    //来客户端·集合添加可以写出的方法
                    allOut.add(pw);//目的：客户端，写出消息--分发消息
                }
            } catch (IOException e) {
                System.out.println("客户端断开链接了~");
            }
        }
        //同步：broadcastMessage()消息广播
        //参数：String  message    转发给客户端  文本
        //增强for：目的--来一个客户端--遍历一个客户端，给这个客户端+消息
        private synchronized void broadcastMessage(String message){
            for(PrintWriter pw:allOut){    //遍历各个客户端--写出
                pw.println("消息广播"+message);//写出消息
            }
        }
        private void handlerChat(){
            new Thread(() -> {
                try {
                    String message;
                    while ((message = br.readLine()) != null) {
                        System.out.println(ip + "客户端说：" + message);
                        System.out.println("服务器请输入：");
                        //理解：客户端写出--服务器读入
                        //    转发消息:客户端写出--服务器读入的内容
                        broadcastMessage(message);
                    }
                } catch (IOException e) {
                    System.out.println("客户端发来的消息断开了~");
                }
            }).start();

            Scanner input = new Scanner(System.in);
            while (true) {
                System.out.println("服务器请输入：");
                String sme = input.nextLine();
                pw.println("服务器说：" + sme);
                if(sme.equalsIgnoreCase("exit")){
                    System.exit(0);
                }
            }
        }
        public void run() {
            try {
                //5.当客户端进行读写功能前，加入登录功能
                //交互问题：服务器---客户端发送：文本
                //        客户端---服务器接收：文本
                //5.1向客户端发送一个提示消息：请输入用户名：
                pw.println("请输入用户名：");
                //5.2客户端输入账号【服务器端读取】
                username=br.readLine();
                pw.println("请输入密码：");
                password=br.readLine();
                //5.3 服务器端:判断账号
                //5.4 用户可能有很多个，封装方法  Login:判断账号密码
                if(Login(username,password)) {
                    pw.println("登陆成功，欢迎进入姜哲牌聊天室");
                    //聊天功能     封装方法：读写功能     handlerChat
                    handlerChat();
                }else{
                    pw.println("登陆失败，用户名或密码错误!");
                    socket.close();//关闭
                }
            } catch(IOException e){
                System.out.println("服务器端读写出问题！");
            }finally {
                //理解:一个客户端已经转发完毕了，移除当前客户端的写出功能
                synchronized (allOut){//目的：移除，刚刚连接客户端
                    allOut.remove(pw);
                }
                if (pw != null)
                    pw.close();
                if (br != null)
                    try {
                        br.close();
                    } catch (IOException e) {
                        System.out.println("读入流断开链接~");
                    }

                if (socket != null && !socket.isClosed()) {
                    System.out.println(ip + "客户端退出服务器链接~");
                    try {
                        socket.close();
                    } catch (IOException e) {
                        System.out.println(ip + "客户端退出连接异常~");
                    }
                }
            }
        }
    }
}
